Added Main class and method