improve output (fix #5)