Add main method + constructor