Fixes bug with excersice 2