Task2 modified

Yet another small bug fixed