Mark code as python