Fixing bug in codes