Updated homework and in class code