cs526 final project, enhancement on project 2