Refined the code

Nu har jag förbättrat koden och snyggat till den avsevärt. Bättre namn,
struktur och organisation!