:gift: Configuration and refactor

お元気ですか @jensechu?  I thought your project was really cool so I wanted to contribute back some changes, I hope you don't mind (feedback always welcomed).
Override the output so that we can intercept the value to assert against.