Removed some unneeded output and made sure to pass in numpy array.