Added few new tests and converted few to data-driven