Adding a Main class for test