bugfix och göra lättare att underhålla koden