Fixes Assignemnt 3