Minor tweak
Fixed issues with previous commit (now correctly interprets return codes from file save / discard)