Added class to handle menu actions