Added first code(currently fails test)