add method to main class