Added test for taking location method