fixed small bug, finished!