Ошибка времени выполнения в python (практический вопрос ранга Хакера)

Question

Итак, я делаю этот вопрос хакерского ранга и получаю ошибки времени выполнения, однако, когда я запускаю его на своей IDE, он отлично работает, как я могу оптимизировать этот код?

#!/bin/python3

import math
import os
import random
import re
import sys

def solve(s):
    list(s)
    x, y = map(str.capitalize, s.split(' '))
    return ' '.join([x, y])

if __name__ == '__main__':
    fptr = open(os.environ['OUTPUT_PATH'], 'w')

    s = input()

    result = solve(s)

    fptr.write(result + '\n')

    fptr.close()