Added basic register functionality