Mer refaktorering og bugfix.