Updated some small code