Added answer to problem 1

I'm totally new to python, so just playing around trying to figure out
its syntax and semantics