added new functions from lab class