small changes DONE