align code correctly