Added styling and updated to jquery instead of plain javascript