Move checks to class

Signed-off-by: Igor Gnatenko <i.gnatenko.brain@gmail.com>