Add an option to show the staged diff