Code organization

Signed-off-by: Fadion Dashi <jonidashi@gmail.com>