Corrected code, added grid
small