Added basic Main class code