Updated as per course project instructions.