My code from working through Learn Python The Hard Way online