Research project (python) | Engineering homework help