added methods to main class