Fix errors and add error styling

Fixes #2 and #3.