Added a small set of tests