small fixes
fixed small changes