update ch4 with more comments

os/src/task/context.rs

@@ -1,6 +1,8 @@
+//! Implementation of [`TaskContext`]
 use crate::trap::trap_return;
 
 #[repr(C)]
+/// task context structure containing some registers
 pub struct TaskContext {
     ra: usize,
     sp: usize,