Add possibility to change proton direction.