Small changes for end of project