Make this code more flexible