Created a test class since I'm having some compilation problems