added more tests :)