Added class component