Small script for testing codeforces problems