Forgot to add main class to initial commit