Fixing 01 exercise