Add location_mock

Create location_mock to replace all the mocks in the fuctions.
Better formatting
Create generator
Add test for when number equal number given