Add 2nd problem source code