[WIP] Implement unit testing