Added first test
Added a couple more tests