code optimization by Varun sir