minor changes :)