added python part to homework