adding in the short cluster thing and update the other python codes