updated - minor bugs

updated few bugs