practicing some numpy and matplotlib stuff