Fix for homework 4.3