layout minor change
module instead of class