Updated main class and added some files.