Fixed #1. Switched to numpy arrays.