made the code fancy