Code tweaking

cause why not.