Small tweaking, fixed the main class location

Signed-off-by: Kohle <kohle@mrkohle.com>