Made small changes :bear: