Just some small fixes

Made things a little clearer