small changes I wanted