minor changes
Dunno what have I done