Fix av buggar samt kodförbättring.