Bug fix + css for error