Add bunch of tests