Minor point: We generally don't mix standard library imports with other imports, so `numpy` should probably go at the end, after a blank line.