A few lil changes