python: correct answers to the first four and a prime module