Improvements for piece api/tests

Update isn't working yet