Updated styles, better working prototype