getting used to numpy