1from data.states.main_menu import main_menu
 2from data.states.town import town
 3from data.states.castle import castle
 4
 5__author__ = 'justinarmstrong'
 6
 7from . import setup, tools
 8from . import constants as c
 9
10def main():
11    """Add states to control here"""
12    run_it = tools.Control(setup.ORIGINAL_CAPTION)
13    state_dict = {c.TOWN: town.Town(c.TOWN),
14                  c.MAIN_MENU: main_menu.Menu(c.MAIN_MENU),
15                  c.CASTLE: castle.Castle(c.CASTLE)}
16
17    run_it.setup_states(state_dict, c.MAIN_MENU)
18    run_it.main()