Update for functionality

Corrected error.