Add some control code