Corrected other small bug