correcting small bugs