adding potential python solution