add comments in model architecture