My version of Assignment from Google Python Class