Added FloatingActionButton and fixed styling