Minor update to clean up files