Sorry for late reply, but I have now fixed what you asked for. Thanks for using it and suggesting improvements :)
