Viết hàm trong python