Add a test class, will be delete later