Add JS for student view

Also fix a few CSS & JS issues.