Refactoring

Signed-off-by: sajith <sajithshn@gmail.com>