Added wrappers for behaviour state