[#10] Improved the entity test. Uses messages for input handling.