Improvements to 'try it' functionality