Added more files that are needed for the assignment