Mindre endringer + ryddet i kode