Minor update to keep all necessary files