Add in lock / unlock functionality