add comments
continue working on it