added grid.draw to be imported
move comments
fixed layout issue