small correctinos
small bug correctinos