A little refactor of context menu to make it a bit more concise