How To Write Algorithm For Classes In Python