Code review and possible small refactor of frontend

@Adam_Soltys we should go over your frontend code and review it.