Add code for the Command Pattern