Add code for two exercises