Added styling and fixed the setup type