cleaned up a little bit

sorry for this code