Optimized code, fixed bugs, completed save answer functionality