Added more comments
Added more code