updated code..minor fixes