Second assignment correction 2