small changes (forgot name :( )