def main():
    speed = input()
    clock = pygame.time.Clock()
    while GameOn:
        clock.tick(speed)
        .
        .
        update_screen(snake, fruit ...)