minor cleanup in Lab 3