[Python] Simple A problems