Fixing typo in class and fixing margin