Python code for Kaggle connectomix competition