Add main class boilerplate for testing purposes