Customize scripts to work with menu