change somethign in the main class :P

Signed-off-by: callum <callumt47@hotmail.co.uk>