small updates to exercise code