small code fixes
small fixes