Please follow the standard coding structure and try to optimize the code as much as possible. Make sure to add comments to make code easily understandable to begineers.
URL : https://github.com/chaudharypraveen98/Competitive-Programming/tree/master/Python/Hackerrank/Write%20a%20function