Added main method for testing!