Generate Python code from problem descriptions
Generate code by infilling between given prefix and suffix