Improve student's answer checking UI