switching to python for the exercise