Testing & tweeking

Testat de nyimplementerade funktionerna och justerat det som behövde
korrigeras.