CLN: Use np.arange instead of list(range)
CLN: Remove whitespace