Unittest writing

unit tests good to go!