Added main to bottom of class