Cleaned up tests file