How to return proper value from a class method