Lazy load entity and repository + presence validation of entity and repo on load

As title
```ruby
return @entity unless klass
@entity = klass
```

what do you think about this?