Tog bort lite

Tog bort och la till lite kod.