Add code for new feature