Thanks a lot Kaspars! This is exactly what I need. Currently using it in my functions.php with some modifications.