Gjorde lite ändringar med interface, hoppas jag tänkte rätt