WIP : Adding unit tests