Improved Self Attention Mechanism