main class fixed

Signed-off-by: Thomas Schubert <funthomas424242@gmail.com>