Added some more specific tests and fixed policy gliph painter to work with smaller states.