Added some tests

Examples of unit testing