updated code and works but isn't styled the same