minor changes

Egy picit szórakoztam a kódoddal