first python code homework