Add identity functionality