Just made some little changes