Added back to main link with less lines of code