Fixed homework from class003