fixed compile error
small edits