Udacity homework, not 100% correct but close need to qork through it more