Found a couple of small issues that bugged me, so tiny update.