moved behaviour class to custom code