added simple visualzation

Replaced get_state with a more intuitive visualization. Obviously there's a lot of room for improvement here, but it's a start.