Three simple changes

Just handy for me. If you like these couple lines of code, please consider merging. Thanks for this simple but very useful program.