Adding init functionality