Add in functionality for link highlighting