Add control functionality and logic