Prism Algorithm Gfg Practice Python