made fixes to code