Add code for problem #11