361: rearrange code