added grid world code