updated based on comment from @takluyver