Removed unnecessary imports
Fixed method call on Integer