Answer:

Not OK. When Player A replaced his ball, Rule 22-2 was satisfactorily completed. Although the player’s ball-marker had not been lifted, the ball was properly in play and there was no Rule which permitted purposely touching the ball for alignment. Player A incurs a one-stroke penalty under Rule 18-2a(i). [The player could have aligned the trademark as he replaced the ball.]