Fixed task four from test