took out explicit numpy import (not needed)