Adding control behavior