Added two behaviour tests