Improve the bot tool