made the code not as shit

still kinda shit though