added in class file and updated main