Replaced csv with pickle