Update to work with processing 2