Rafactored to true DI =)