Refactor using user_id()
Refactor and make class