Simplifying the code a bit