made some small fixes
made some small fixes