Solved pickle problem and added test