slight frontend changes
First iteration of code