Functionality

Add correct answer check