Add basic register functionality