First Complete python code

Doing some codeacademy