[WIP] Added small test suite