Added code for algo lab