improved comments, removed unnecessary imports