Tog bort lite kod

och la till lite kod