Added the ability to add a command end custom callback