Fixed bugin HW5