Made a few tweaks to the add_entry script