Remove main method

Close #4