Small refuctoring

I am bored