minor error, functionality updates