Getting those shapes right: Beginners guide to NumPy