Added tests - re-added lombok @Data to registers (previously removed in error)