adding some code for the mocks
blarpy edits