pyth a b = a * a + b * b
main = print $ pyth 3 4