Added support for new files -duh