Added in class code alongs