Added small comments and documentation