import gravity_mod as gfx
import time
import random

# Method to update grid
def updateGrid(grid):
for x in range(60):
   for y in range(58,-1,-1):
         if grid[x][y] == "SAND" and grid[x][y+1] == "":
            grid[x][y] = ""
            grid[x][y+1] = "SAND"
         elif grid[x][y] == "SAND" and grid[x][y+1] == "WATER":
            grid[x][y] = "WATER"
            grid[x][y+1] = "SAND"
         elif grid[x][y] == "WATER" and grid[x][y+1] == "":
            grid[x][y] = ""
            grid[x][y+1] = "WATER"
         elif grid[x][y] == "WATER" and grid[x][y+1] != "":
            choices = []
            if x > 0 and grid[x-1][y] == "":
               choices.append((x-1,y))
            if x < 59 and grid[x+1][y] == "":
               choices.append((x+1,y))
            if len(choices) > 0:
               x2, y2 = random.sample(choices, 1)[0]
               grid[x][y] = ""
               grid[x2][y2] = "WATER"

# Method to get events from window
def getEvents(win, grid):
for event in win.get_events():
   if event[0] in ["STONE","SAND","WATER"]:
         x = int(event[1]/10)
         y = int(event[2]/10)
         if grid[x][y] == '':
            grid[x][y] = event[0]
   elif event == "CLEAR":
         grid = [[""]*60 for i in range(60)]
return grid

# Method to draw grid
def drawGrid(grid):
for x in range(60):
   for y in range(60):
         if grid[x][y] == "STONE":
            win.draw_rect(x*10,y*10,(x+1)*10,(y+1)*10,"Gray")
         elif grid[x][y] == "SAND":

DoubleMega, the company that you work for needs a more advanced sea...