Small changes to Project 2.