alignment to @nicolapcweek94's code

in the meanwhile, code was modified. aligning.