Work in progress, main class implemented