added some little unit testing