cleaned code up a bit
updated module name