Add behaviour class