updated grid code to be more clean, and have dedicated setup and cleanup functionalities. hopefully code will stay cleaner