Lil modifications!! :)