Changes so we can handle other file types apart from class files