Added more comments and fixed some imports