Few lines

Added 2 classes