Update to include new files