Add problem definition to 2sum py