Adding in the UI part of restrictions