using cols is better
abstracted mock to mock.py
make result better