Use pickle instead of yaml