Add testcases to assert the behaviour