minor changes :)

nearly finished :D