Added entity.shape method to the generator.