added option to get classes