cleaned up session 5 homework