fixed small bug
fixed bug with classes