Minor Update

Hi All,

Just a small update fixing a couple minor bugs.

As usual, if you find any bugs, please send to [email protected] and I will jump on them.

Thanks