small bug fixes
fixed bugs