exercise 2 in pyplasm