Fix some issues on main screen

1 files changed, 14 insertions, 18 deletions

diff --git a/modern/src/MainToolbar.js b/modern/src/MainToolbar.js
index e8294ab..1420135 100644
--- a/modern/src/MainToolbar.js
+++ b/modern/src/MainToolbar.js
@@ -1,5 +1,4 @@
-import React from 'react';
-import PropTypes from 'prop-types';
+import React, { Component } from 'react';
 import { withStyles } from '@material-ui/core/styles';
 import AppBar from '@material-ui/core/AppBar';
 import Toolbar from '@material-ui/core/Toolbar';
@@ -9,22 +8,23 @@ import IconButton from '@material-ui/core/IconButton';
 import MenuIcon from '@material-ui/icons/Menu';
 
 const styles = {
-  root: {
-    flexGrow: 1,
-  },
   flex: {
-    flexGrow: 1,
+    flexGrow: 1
   },
   menuButton: {
     marginLeft: -12,
-    marginRight: 20,
+    marginRight: 20
   },
 };
 
-function MainToobar(props) {
-  const { classes } = props;
-  return (
-    <div className={classes.root}>
+class MainToobar extends Component {
+  constructor(props) {
+    super(props);
+  }
+
+  render() {
+    const { classes } = this.props;
+    return (
       <AppBar position="static">
         <Toolbar>
           <IconButton className={classes.menuButton} color="inherit" aria-label="Menu">
@@ -33,15 +33,11 @@ function MainToobar(props) {
           <Typography variant="title" color="inherit" className={classes.flex}>
             Traccar
           </Typography>
-          <Button color="inherit">Login</Button>
+          <Button color="inherit">Logout</Button>
         </Toolbar>
       </AppBar>
-    </div>
-  );
+    );
+  }
 }
 
-MainToobar.propTypes = {
-  classes: PropTypes.object.isRequired,
-};
-
 export default withStyles(styles)(MainToobar);