Added basic Class and Main code