from leilighet import Leilighet
from bygaard import Bygaard

bygaard = Bygaard()

l1 = Leilighet(3, 304)
l2 = Leilighet(2, 201)

bygaard.leggTilLeilighet(l1)
bygaard.leggTilLeilighet(l2)

hjem = bygaard.finnLeilighet(201)
if hjem != None:
    print("Jeg bor i {}. etasje i rom nr. {}".format(\
        hjem.hentEtasje(), hjem.hentRomnummer()))