Got junit tests working