Make the code a bit more flexible