I assume the main method is for testing purposes. Can you remove it?