FIxade till koden..igen..

Uppdaterade koden, inte senaste som fanns på github..