Added remove code command
Added conditional to level script