small changes not done yet

small changes not done yet