Fixed slight bugs :))))