Updated tests file