studied numpy, barely started exercise 1