Added two-liner class