Add more inline documentation