super small python solution