removing redundant np.newaxis in line.py example