Small changes
Fix small bug