Minor cleanup of Scientific Python 2 exercise