1 files changed, 15 insertions, 7 deletions

diff --git a/cmd/web/js/poloniex.jsx b/cmd/web/js/poloniex.jsx
index c0b68c7..482dbb6 100644
--- a/cmd/web/js/poloniex.jsx
+++ b/cmd/web/js/poloniex.jsx
@@ -1,5 +1,4 @@
 import Api from './api.js';
-import classNames from 'classnames';
 import React from 'react';
 
 class PoloniexController extends React.Component {
@@ -94,6 +93,15 @@ class PoloniexController extends React.Component {
   }
 }
 
+class CurrencyLogo extends React.Component {
+  render = () => {
+    return <img className="currency-logo"
+                src={'/public/icons/black/' + this.props.currency.toLowerCase() + '.svg' }
+                title={this.props.currency}
+                alt={this.props.currency} />;
+  }
+}
+
 class PoloniexBalance extends React.Component {
   constructor(props) {
     super(props);
@@ -106,19 +114,19 @@ class PoloniexBalance extends React.Component {
     if (this.props.balanceValue !== null) {
 
       var balance = Object.keys(this.props.balance).map(function(currency) {
-        return <div key={currency}><i className={classNames('cc', currency)}></i> {this.props.balance[currency]}</div>;
+        return <div key={currency}>
+                 <CurrencyLogo currency={currency} /> {this.props.balance[currency]}
+               </div>;
       }.bind(this));
 
       dashboard =
         <div className="row">
-          <div className="col-4 align-self-center h-100">
-            <div>
+          <div className="col-4 align-self-center h-100 balances">
               {balance}
-            </div>
           </div>
           <div className="offset-1 col-7 h-100 align-self-center">
             <div className="text-center">
-              Balance ({this.props.balanceCurrency}): <span>{this.props.balanceValue} <i className={classNames('cc', this.props.balanceCurrency)}></i></span>
+              Balance ({this.props.balanceCurrency}): <span>{this.props.balanceValue}</span><CurrencyLogo currency={this.props.balanceCurrency} />
             </div>
           </div>
         </div>;
@@ -133,7 +141,7 @@ class PoloniexBalance extends React.Component {
 
     return (
       <div className="row">
-        <div className="box offset-2 col-8">
+        <div className="box offset-2 col-8 portfolio">
           <div className="row">
             <div className="col-4">Portfolio</div>
           </div>