Fixed task1 bug