Added a repository
Now using dependency injection
Removed unused files