Return Triangle Python