Add some controls yo