Refactor method

Signed-off-by: Mihaela Martinas <martinas.mihaela21@gmail.com>