Use repository class

Now we can use custom repository find() method