- fixed a couple minor bugs