Adding tests to check entity creation