Updated comments on python classes