Add grid class for object oriented life in python