Just some small changes :-)