Ignore class files
Add more tests and refactor test class