#78 bug fixes