Dynamic Array In Python Hackerrank Test