Add styles
Fix tests