Updated the code to work with a PRismino