Fixed some little issues

Fixed some stuff