Finally got controls working, still need some tweaking