Adding new efficiency problem python code