minor code changed