Code documentation using python