Fixed Question4 for assigment 2