removed all class files and improved the code